package com.thinkcms.core.utils;
import com.thinkcms.core.constants.Constants;
import lombok.extern.slf4j.Slf4j;
import java.io.*;
import java.util.Enumeration;

@Slf4j
public class ZipUtil {


    /**
     * 使用GBK编码可以避免压缩中文文件名乱码 
     */
    private static final String CHINESE_CHARSET = "GBK";

    /**
     * 文件读取缓冲区大小 
     */
    private static final int CACHE_SIZE = 1024;

    private ZipUtil(){
        // 私用构造主法.因为此类是工具类.
    }

    /**
     * <p> 
     * 压缩文件 
     * </p> 
     *
     * @param sourceFolder 需压缩文件 或者 文件夹 路径
     * @param zipFilePath 压缩文件输出路径 
     * @throws Exception
     */
    public static void zip(String sourceFolder, String zipFilePath) throws Exception {
        if(zipFilePath.length()>=3){
            String suffix =FileUtil.getSuffix(zipFilePath);
            if(!Constants.DEFAULT_ZIP_SUFFIX.equals(Constants.DOT+suffix)){
                zipFilePath+= Constants.DEFAULT_ZIP_SUFFIX;
            }
        }
        log.debug("开始压缩 ["+sourceFolder+"] 到 ["+zipFilePath+"]");
        OutputStream out = new FileOutputStream(zipFilePath);
        BufferedOutputStream bos = new BufferedOutputStream(out);
        org.apache.tools.zip.ZipOutputStream zos = new org.apache.tools.zip.ZipOutputStream(bos);
        // 解决中文文件名乱码
        zos.setEncoding(CHINESE_CHARSET);
        File file = new File(sourceFolder);
        String basePath = null;
        if (file.isDirectory()) {
            basePath = file.getPath();
        } else {
            basePath = file.getParent();
        }
        zipFile(file, basePath, zos);
        zos.closeEntry();
        zos.close();
        bos.close();
        out.close();
        log.debug("压缩 ["+sourceFolder+"] 完成！");
    }

    /**
     * <p>
     * 压缩文件
     * </p>
     *
     * @param sourceFolders 一组 压缩文件夹 或 文件
     * @param zipFilePath 压缩文件输出路径
     * @throws Exception
     */
    public static void zip(String[] sourceFolders, String zipFilePath) throws Exception {
        OutputStream out = new FileOutputStream(zipFilePath);
        BufferedOutputStream bos = new BufferedOutputStream(out);
        org.apache.tools.zip.ZipOutputStream zos = new org.apache.tools.zip.ZipOutputStream(bos);
        // 解决中文文件名乱码
        zos.setEncoding(CHINESE_CHARSET);

        for (int i = 0; i < sourceFolders.length; i++) {
            log.debug("开始压缩 ["+sourceFolders[i]+"] 到 ["+zipFilePath+"]");
            File file = new File(sourceFolders[i]);
            String basePath = null;
            basePath = file.getParent();
            zipFile(file, basePath, zos);
        }

        zos.closeEntry();
        zos.close();
        bos.close();
        out.close();
        log.debug("压缩 "+sourceFolders+" 完成！");
    }

    /**
     * <p>
     * 递归压缩文件
     * </p>
     *
     * @param parentFile
     * @param basePath
     * @param zos
     * @throws Exception
     */
    private static void zipFile(File parentFile, String basePath, org.apache.tools.zip.ZipOutputStream zos) throws Exception {
        File[] files = new File[0];
        if (parentFile.isDirectory()) {
            files = parentFile.listFiles();
        } else {
            files = new File[1];
            files[0] = parentFile;
        }
        String pathName;
        InputStream is;
        BufferedInputStream bis;
        byte[] cache = new byte[CACHE_SIZE];
        for (File file : files) {
            if (file.isDirectory()) {
                log.debug("目录："+file.getPath());
                basePath = basePath.replace('\\', '/');
                if(basePath.substring(basePath.length()-1).equals("/")){
                    pathName = file.getPath().substring(basePath.length()) + "/";
                }else{
                    pathName = file.getPath().substring(basePath.length() + 1) + "/";
                }

                zos.putNextEntry(new org.apache.tools.zip.ZipEntry(pathName));
                zipFile(file, basePath, zos);
            } else {
                pathName = file.getPath().substring(basePath.length()) ;
                pathName = pathName.replace('\\', '/');
                if(pathName.substring(0,1).equals("/")){
                    pathName = pathName.substring(1);
                }
                log.debug("压缩："+pathName);
                is = new FileInputStream(file);
                bis = new BufferedInputStream(is);
                zos.putNextEntry(new org.apache.tools.zip.ZipEntry(pathName));
                int nRead = 0;
                while ((nRead = bis.read(cache, 0, CACHE_SIZE)) != -1) {
                    zos.write(cache, 0, nRead);
                }
                bis.close();
                is.close();
            }
        }
    }

    /**
     * 解压zip文件
     *
     * @param zipFileName
     *            待解压的zip文件路径，例如：c:\\a.zip
     *
     * @param outputDirectory
     *            解压目标文件夹,例如：c:\\a\
     */
    public static void unZip(String zipFileName, String outputDirectory)
            throws Exception {
        log.debug("开始解压 ["+zipFileName+"] 到 ["+outputDirectory+"]");
        org.apache.tools.zip.ZipFile zipFile = new org.apache.tools.zip.ZipFile(zipFileName);

        try {

            Enumeration<?> e = zipFile.getEntries();

            org.apache.tools.zip.ZipEntry zipEntry = null;

            createDirectory(outputDirectory, "");

            while (e.hasMoreElements()) {

                zipEntry = (org.apache.tools.zip.ZipEntry) e.nextElement();

                log.debug("解压：" + zipEntry.getName());

                if (zipEntry.isDirectory()) {

                    String name = zipEntry.getName();

                    name = name.substring(0, name.length() - 1);

                    File f = new File(outputDirectory + File.separator + name);

                    f.mkdir();

                    log.debug("创建目录：" + outputDirectory + File.separator + name);

                } else {

                    String fileName = zipEntry.getName();

                    fileName = fileName.replace('\\', '/');

                    if (fileName.indexOf("/") != -1) {

                        createDirectory(outputDirectory, fileName.substring(0,
                                fileName.lastIndexOf("/")));

                        fileName = fileName.substring(
                                fileName.lastIndexOf("/") + 1,
                                fileName.length());

                    }

                    File f = new File(outputDirectory + File.separator
                            + zipEntry.getName());

                    f.createNewFile();

                    InputStream in = zipFile.getInputStream(zipEntry);

                    FileOutputStream out = new FileOutputStream(f);

                    byte[] by = new byte[1024];

                    int c;

                    while ((c = in.read(by)) != -1) {

                        out.write(by, 0, c);

                    }

                    in.close();

                    out.close();

                }

            }
            log.debug("解压 ["+zipFileName+"] 完成！");

        } catch (Exception ex) {

            System.out.println(ex.getMessage());

        } finally {
            zipFile.close();
        }

    }


    public static void unZip(File file, String outputDirectory)
            throws Exception {
        log.debug("开始解压 ["+file+"] 到 ["+outputDirectory+"]");
        org.apache.tools.zip.ZipFile zipFile = new org.apache.tools.zip.ZipFile(file,"utf-8");

        try {

            Enumeration<?> e = zipFile.getEntries();

            org.apache.tools.zip.ZipEntry zipEntry = null;

            createDirectory(outputDirectory, "");

            while (e.hasMoreElements()) {

                zipEntry = (org.apache.tools.zip.ZipEntry) e.nextElement();

                log.debug("解压：" + zipEntry.getName());

                if (zipEntry.isDirectory()) {

                    String name = zipEntry.getName();

                    name = name.substring(0, name.length() - 1);

                    File f = new File(outputDirectory + File.separator + name);

                    f.mkdir();

                    log.debug("创建目录：" + outputDirectory + File.separator + name);

                } else {

                    String fileName = zipEntry.getName();

                    fileName = fileName.replace('\\', '/');

                    if (fileName.indexOf("/") != -1) {

                        createDirectory(outputDirectory, fileName.substring(0,
                                fileName.lastIndexOf("/")));

                        fileName = fileName.substring(
                                fileName.lastIndexOf("/") + 1,
                                fileName.length());

                    }

                    File f = new File(outputDirectory + File.separator
                            + zipEntry.getName());

                    f.createNewFile();

                    InputStream in = zipFile.getInputStream(zipEntry);

                    FileOutputStream out = new FileOutputStream(f);

                    byte[] by = new byte[1024];

                    int c;

                    while ((c = in.read(by)) != -1) {

                        out.write(by, 0, c);

                    }

                    in.close();

                    out.close();

                }

            }
            log.debug("解压 ["+file+"] 完成！");

        } catch (Exception ex) {

            System.out.println(ex.getMessage());

        } finally {
            zipFile.close();
        }

    }

    /**
     * 创建目录
     * @author hezhao
     * @Time   2017年7月28日 下午7:10:05
     * @param directory
     * @param subDirectory
     */
    private static void createDirectory(String directory, String subDirectory) {

        String dir[];

        File fl = new File(directory);

        try {

            if (subDirectory == "" && fl.exists() != true) {

                fl.mkdir();

            } else if (subDirectory != "") {

                dir = subDirectory.replace('\\', '/').split("/");

                for (int i = 0; i < dir.length; i++) {

                    File subFile = new File(directory + File.separator + dir[i]);

                    if (subFile.exists() == false)

                        subFile.mkdir();

                    directory += File.separator + dir[i];

                }

            }

        } catch (Exception ex) {

            System.out.println(ex.getMessage());

        }

    }

    /**
     * 无需解压直接读取Zip文件和文件内容
     * @author hezhao
     * @Time   2017年7月28日 下午3:23:10
     * @param file 文件
     * @throws Exception
     */
    public static void readZipFile(String file) throws Exception {
        java.util.zip.ZipFile zipFile = new java.util.zip.ZipFile(file);
        InputStream in = new BufferedInputStream(new FileInputStream(file));
        java.util.zip.ZipInputStream zin = new java.util.zip.ZipInputStream(in);
        java.util.zip.ZipEntry ze;
        while ((ze = zin.getNextEntry()) != null) {
            if (ze.isDirectory()) {
            } else {
                log.info("file - " + ze.getName() + " : "
                        + ze.getSize() + " bytes");
                long size = ze.getSize();
                if (size > 0) {
                    BufferedReader br = new BufferedReader(
                            new InputStreamReader(zipFile.getInputStream(ze)));
                    String line;
                    while ((line = br.readLine()) != null) {
                        System.out.println(line);
                    }
                    br.close();
                }
                System.out.println();
            }
        }
        zin.closeEntry();
    }


    public static void main(String[] args) throws Exception {
        try {
//          readZipFile("D:\\new1\\文字.zip");

            //压缩文件
            //String sourceFolder = "C:\\Users\\LG\\Desktop\\费用审批记录查询";
            //String zipFilePath = "C:\\Users\\LG\\Desktop\\费用审批记录查询.zip";
            //ZipUtil.zip(sourceFolder, zipFilePath);

            //压缩文件夹
//          String sourceFolder = "D:/fsc1";
//          String zipFilePath = "D:/fsc1.zip";  
//          ZipUtil.zip(sourceFolder, zipFilePath);

            //压缩一组文件
//          String [] paths = {"D:/新建文本文档.txt","D:\\FastStoneCapturecn.zip","D:/new1"};
//          zip(paths, "D:/abc.zip");

          unZip("C:\\Users\\LG\\Desktop\\费用审批记录查询.zip", "C:\\Users\\LG\\Desktop\\费用审批记录查询1");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}
